Almost there on remove button fix
Last line. Fixes #346