fixing slight bug