Fixed another feature