fixed bug in question