it is working , but with a bug