Fixes #5 - (i) button not working.