Minor Bug Fix