Fixed little problem