Updated a bug. this is a fix