fixed bug in answer