Minor bug fix

Still bugged...