Fix button part 2