Bug Fix

- fixed if() statement finally (I think)