Quick code fix