Small bugg fix