added bug back