Better handling of flex unit tests