Fix directory parameters in __main__