slight bug fixed