Removed post import.
Fixed tests