fix(*): code is not removed from files with custom extension

Closes #1