Small bugg fixed