fixing compilation
added in support for removing certain units