Quick fix to make buttons work