Fixed tests

Added missing files