Minor Bug Fix!